package lx;

import java.io.File;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

public class demo6 {
    //1.定义集合用来统计
    static HashMap<String, Integer> hm = new HashMap<>();

    public static void main(String[] args) {
          /*
            需求：统计一个文件夹中每种文件的个数并打印。（考虑子文件夹）
            打印格式如下：
            txt:3个
            doc:4个
            jpg:6个
        */
        File file = new File("D:\\Java学习\\aaa");
        HashMap<String, Integer> hm = getCount(file);
        System.out.println(hm);
    }

    /*
     * 作用：
     *       统计一个文件夹中每种文件的个数
     * 参数：
     *       要统计的那个文件夹
     * 返回值：
     *       用来统计map集合
     *       键：后缀名 值：次数
     *
     *       a.txt
     *       a.a.txt
     *       aaa（不需要统计的）
     *
     *
     * */
    public static HashMap<String, Integer> getCount(File src) {
        //2.进入src文件夹
        File[] files = src.listFiles();
        //3.遍历数组
        for (File file : files) {
            //4.判断，如果是文件，统计
            if (file.isFile()) {
                //a.txt
                String name = file.getName();
                String[] arr = name.split("\\.");
                //这里的arr.length就是以.为分割，文件的长度，如1.txt的就是 1和txt，数组长度为2
                //为3的就是 1.txt.obj ---1 和txt 和obj
                if (arr.length >= 2) {
                    //获取结尾的后缀
                    String endName = arr[arr.length - 1];
                    //在hm集合里查看这个后缀有吗
                    if (hm.containsKey(endName)) {
                        //存在
                        int count = hm.get(endName);
                        count++;
                        hm.put(endName, count);
                    } else {
                        //不存在
                        hm.put(endName, 1);
                    }
                }
            } else {
                //5判断，如果是文件夹，统计
                getCount(file);
            }

        }
        return hm;
    }
}
